package 技巧.颜色分类;
/**
 * 题目描述：
 * 颜色分类
 * 给定一个包含红色、白色和蓝色、共 n 个元素的数组 nums ，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。
 * 我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
 * 必须在不使用库内置的 sort 函数的情况下解决这个问题。
 * 输入：nums = [2,0,2,1,1,0]
 * 输出：[0,0,1,1,2,2]
 * 输入：nums = [2,0,1]
 * 输出：[0,1,2]
 * 力扣：https://leetcode-cn.com/problems/sort-colors/
 * 力扣题目编号：75
 * 时间复杂度：O(n)
 * 空间复杂度：O(1)
 * */
/**
 * 解题思路：
 * 创建数组统计各种元素出现的次数
 * 然后分别填入0,1,2即可
 * */

class Solution {
    public void sortColors(int[] nums) {
        int[] cnt = new int[3];
        for (int num : nums){
            cnt[num]++;
        } 
        for(int i = 0;i < cnt[0];i++){
            nums[i] = 0;
        }
        for(int i = cnt[0];i < cnt[0] + cnt[1];i++){
            nums[i] = 1;
        }     
        for(int i = cnt[0] + cnt[1];i < cnt[0] + cnt[1] + cnt[2];i++){
            nums[i] = 2;
        }
    }
 }